package com.ys.otms.utils;

import com.ys.otms.services.impl.YLShipsServiceImpl;
import net.sf.json.JSONObject;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadBase;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.logging.Logger;

/**
 * Created by Administrator on 2018/11/4.
 */
public class UploadUtil {
    private static org.slf4j.Logger logger = LoggerFactory.getLogger(UploadUtil.class);

    public static JSONObject getUpload(HttpServletRequest request) throws ServletException,IOException {
        String imagePaths = "";
        String savePath = request.getSession().getServletContext().getRealPath("/");
        String[] sourceStrArray = savePath.split("webapps");
        savePath = sourceStrArray[0] + "/webapps/excel";
        //savePath = sourceStrArray[0] + "\\webapps\\excel";
        File saveFile = new File(savePath);
        if (!saveFile.exists()) {
            saveFile.mkdir();
        }
        JSONObject jsonObject = new JSONObject();
        String filePath =null;
        String message = "";
        try {
            //使用Apache文件上传组件处理文件上传步骤：
            //1、创建一个DiskFileItemFactory工厂
            DiskFileItemFactory factory = new DiskFileItemFactory();
            //设置工厂的缓冲区的大小，当上传的文件大小超过缓冲区的大小时，就会生成一个临时文件存放到指定的临时目录当中。
            factory.setSizeThreshold(1024 * 100);//设置缓冲区的大小为100KB，如果不指定，那么缓冲区的大小默认是10KB
            //设置上传时生成的临时文件的保存目录
            factory.setRepository(saveFile);
            //2、创建一个文件上传解析器
            ServletFileUpload upload = new ServletFileUpload(factory);
            //监听文件上传进度
            upload.setProgressListener(new ProgressListener() {
                public void update(long pBytesRead, long pContentLength, int arg2) {
//                     System.out.println("文件大小为：" + pContentLength + ",当前已处理：" + pBytesRead);
//                        /*** 文件大小为：14608,当前已处理：4096
//                      文件大小为：14608,当前已处理：7367
//                        文件大小为：14608,当前已处理：11419
//                       文件大小为：14608,当前已处理：14608*/
                }
            });
            //解决上传文件名的中文乱码
            upload.setHeaderEncoding("UTF-8");
            //3、判断提交上来的数据是否是上传表单的数据
            if (!ServletFileUpload.isMultipartContent(request)) {
                //按照传统方式获取数据
                jsonObject.put("imagePaths",imagePaths);
                return jsonObject;
            }
            //设置上传单个文件的大小的最大值，目前是设置为1024*1024字节，也就是1MB
            upload.setFileSizeMax(1024 * 1024*100);
            //设置上传文件总量的最大值，最大值=同时上传的多个文件的大小的最大值的和，目前设置为10MB
            upload.setSizeMax(1024 * 1024 * 100);
            //4、使用ServletFileUpload解析器解析上传数据，解析结果返回的是一个List<FileItem>集合，每一个FileItem对应一个Form表单的输入项
            List<FileItem> list = upload.parseRequest(request);
            for (FileItem item : list) {
                //如果fileitem中封装的是普通输入项的数据
                if (item.isFormField()) {
                    String name = item.getFieldName();
                    //解决普通输入项的数据的中文乱码问题
                    String value = item.getString("UTF-8");
                    //value = new String(value.getBytes("iso8859-1"),"UTF-8");
                    //system.out.println(name + "=" + value);
                    imagePaths = imagePaths + value + ",";
                } else {//如果fileitem中封装的是上传文件
                    //得到上传的文件名称，
                    String filename = item.getName();
                    //System.out.println(filename);
                    if (filename == null || filename.trim().equals("")) {
                        imagePaths = imagePaths + " " + ",";
                        continue;
                    }
                    //注意：不同的浏览器提交的文件名是不一样的，有些浏览器提交上来的文件名是带有路径的，如：  c:\a\b\1.txt，而有些只是单纯的文件名，如：1.txt
                    //处理获取到的上传文件的文件名的路径部分，只保留文件名部分
                    filename = filename.substring(filename.lastIndexOf("\\") + 1);
                    //如果需要限制上传的文件类型，那么可以通过文件的扩展名来判断上传的文件类型是否合法
                    // System.out.println("上传的文件的扩展名是："+fileExtName);
                    //获取item中的上传文件的输入流
                    filePath =savePath+"/"+ filename;
                    InputStream in = item.getInputStream();
                    //得到文件保存的名称
                    imagePaths = imagePaths + filename + ",";
                    logger.info("imagePaths：",imagePaths);
                    //得到文件的保存目录
//                    String realSavePath = makePath(saveFilename, savePath);
                    //创建一个文件输出流
                    FileOutputStream out = new FileOutputStream(savePath + "/" + filename);
                    //创建一个缓冲区
                    byte buffer[] = new byte[1024];
                    //判断输入流中的数据是否已经读完的标识
                    int len = 0;
                    //循环将输入流读入到缓冲区当中，(len=in.read(buffer))>0就表示in里面还有数据
                    while ((len = in.read(buffer)) > 0) {
                        //使用FileOutputStream输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + filename)当中
                        out.write(buffer, 0, len);
                    }
                    //关闭输入流
                    in.close();
                    //关闭输出流
                    out.close();
                    //删除处理文件上传时生成的临时文件
                    //item.delete();
                    message = "SUCCESS";

                }
            }
        } catch (FileUploadBase.FileSizeLimitExceededException e) {
            e.printStackTrace();
            //  request.setAttribute("message", "单个文件超出最大值！！！");
            // request.getRequestDispatcher("/message.jsp").forward(request, response);
            message ="单个文件超出最大值！！！";
        } catch (FileUploadBase.SizeLimitExceededException e) {
            e.printStackTrace();
            // request.setAttribute("message", "上传文件的总的大小超出限制的最大值！！！");
            // request.getRequestDispatcher("/message.jsp").forward(request, response);
            message="上传文件的总的大小超出限制的最大值！！！";
        } catch (Exception e) {
            message = "文件上传失败！";
            e.printStackTrace();
        }
        logger.info("message：",message);
        jsonObject.put("message",message);
        jsonObject.put("fileName",filePath);
        return jsonObject;
    }
}
